import math
import numpy as np

def distance(x1, x2, y1, y2, z1=0, z2=0):
    dis = math.sqrt((x2 - x1) ** 2 + (y2 - y1) ** 2 + (z2 - z1) ** 2)
    return dis